Torch.nn.embedding Explained